Plotting Plays, Python Data (Standardized Direction and Location)

0.1 Topic

This file

0.2 Set Up

Packages/Files

library(tidyverse)
library(magrittr)

Load Data

loadData <- function(fileName = "train.csv", ...) {
  file = paste("../input", fileName, sep="/")
  data.table::fread(file, ...) %>% 
    as_tibble()
}
df = loadData("trainClean_py.csv")
df %<>%
  mutate(
    X_end = S*cos((90-Dir)*pi/180) + X, 
    Y_end = S*sin((90-Dir)*pi/180) + Y
  )
dfFeatures = loadData("features_py.csv")

1 Plot code from rescale file

Taken from https://www.kaggle.com/statsbymichaellopez/nfl-tracking-wrangling-voronoi-and-sonars#Dealing-with-player-tracking-data

options(repr.plot.width=10, repr.plot.height=6)
samp_play <- "20170910001102"
plotPlay = function(dfPlay, features=NULL, run=F) {
  rusherSpeed = dfPlay %>% 
    filter(NflId == NflIdRusher) %>% 
    pull(S)
  p = dfPlay %>% 
    ggplot(aes(X, Y, color = OnOffense))  + 
    geom_point(size = 2) + 
    geom_segment(aes(x = X, y = Y, xend = X_end, yend = Y_end), 
                 arrow = arrow(length = unit(.5,"cm"))) + 
    geom_point(data = filter(dfPlay, NflId == NflIdRusher), 
               pch=21, size = 1.5, fill = "black") +
    # geom_point(data = filter(dfPlay, Position == "QB"), size = 2, fill = "blue") +
    scale_colour_brewer(palette = "Set2")+ 
    scale_fill_brewer(palette = "Set2")+ 
    # geom_vline(aes(xintercept=0), color="grey") +
    # geom_vline(aes(xintercept=10), color="grey") +
    geom_vline(aes(xintercept = LineOfScrimmage), colour = "black", lty = 2) + 
    coord_cartesian(ylim=c(0,160/3), expand=F) + #xlim = c(-10,110), 
    labs(
      x = "Distance from offensive team's own end zone", 
      y = "Y", 
      title = paste0("PlayId ", pull(dfPlay, PlayId)), 
      subtitle = paste("Yards", dfPlay$Yards, "; ", "Down", dfPlay$Down)
    ) + 
    theme_bw(14) + 
    theme(panel.grid.minor = element_blank(), panel.grid.major.y =element_blank())
  if (run && !is.null(features)) {
    dfC = circleDF(c(features$LineOfScrimmage, features$Rusher_Gap_Center),
                   features$Rusher_Gap_Radius*rusherSpeed)
    p + 
      geom_path(data = dfC, aes(x, y), color="gold") +
      labs(subtitle = paste("Yards", dfPlay$Yards, "; ", "Down", dfPlay$Down, "; ",
                            "NPlayers", features$Rusher_Gap_NPlayers, "; ",
                            "OpenSize", round(features$Rusher_Gap_OpenSize, 2)))
  } else {
    p
  }
}
circleDF = function(center = c(0,0), r = 1, npoints = 100){
    tt <- seq(0, 2*pi, length.out = npoints)
    xx <- center[1] + r * cos(tt)
    yy <- center[2] + r * sin(tt)
    return(data.frame(x = xx, y = yy))
}
dfPlay = filter(df, PlayId == samp_play)
features = filter(dfFeatures, PlayId == samp_play)
plotPlay(dfPlay, features, run=T)

2 Looking at Plays

First, categorize runs: long, medium, short, bad

ggplot(df) +
  geom_density(aes(Yards))

summary(df$Yards)

   Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
-14.000   1.000   3.000   4.212   6.000  99.000 

df %<>%
  mutate(YardsCategory = cut(Yards, breaks = c(-Inf, 0, 3, 6, Inf), 
             labels=c("bad", "short", "med", "long")))
table(df$YardsCategory)

   bad  short    med   long 
103532 184888 117106 104236 

PlayId_long = df %>% 
  filter(YardsCategory == "long") %>% 
  pull(PlayId)
PlayId_med = df %>% 
  filter(YardsCategory == "med") %>% 
  pull(PlayId)
PlayId_short = df %>% 
  filter(YardsCategory == "short") %>% 
  pull(PlayId)
PlayId_bad = df %>% 
  filter(YardsCategory == "bad") %>% 
  pull(PlayId)

Plot several runs from vector of ids

plotPlays = function(ids, n=5) {
  for (index in sample.int(length(ids), min(n, length(ids)))) {
    id = ids[index]
    dfPlay = filter(df, PlayId == id)
    features = filter(dfFeatures, PlayId == id)
    p = plotPlay(dfPlay, features, run=T)
    print(p)
  }
}

2.1 Individual Plays

plotPlays("20181230154157")

20181230154157

[1] 2.018123e+13

2.2 Long Runs

plotPlays(PlayId_long)

2.3 Medium Runs

plotPlays(PlayId_med)

2.4 Short Runs

plotPlays(PlayId_short)

2.5 Bad Runs

plotPlays(PlayId_bad)

2.6 To Edge

edgePlays = dfFeatures %>% 
  filter(Rusher_Gap_ToEdge == 1) %>% 
  pull(PlayId)
plotPlays(edgePlays)